#pragma once
#include <iostream>
#include "compiler.hpp"
#include "runner.hpp"
#include "../comm/util.hpp"
#include "../comm/log.hpp"
#include <jsoncpp/json/json.h>

namespace NS_CompileAndRun
{
    using namespace NS_Compiler;
    using namespace NS_Runner;
    using namespace NS_Util;
    using namespace NS_Log;

    class CompileAndRun
    {
    public:
        /**************************************************************
         * 输入json串：
         *   code    : 用户提交代码
         *   input   : 用户提交代码的标准输入内容，不作处理以待扩展
         *   cpu_lim : 时间复杂度
         *   mem_lim : 空间复杂度
         * {"code": "#include...", "input": "", "cpu_lim": 1, "mem_lim": 10240}
         *
         * 输出json串：
         *   必填：
         *   status  : 状态码
         *   reason  : 结果原因
         *   选填：
         *   stdout  : 用户提交代码的运行正确结果
         *   stderr  : 用户提交代码的运行错误结果
         * {"status": "0", "reason": "", "stdout": "", "stdin": ""}
        ***************************************************************/
        static void Start(const std::string& in_json, std::string* out_json)
        {
            // 反序列化
            Json::Value in_value;
            Json::Reader reader;
            reader.parse(in_json, in_value); // 解析in_json

            std::string code = in_value["code"].asString();
            std::string input = in_value["input"].asString();
            int cpu_limit = in_value["cpu_lim"].asInt();
            int mem_limit = in_value["mem_lim"].asInt();

            int status_code = 0;
            int ret_code = 0;
            std::string file_name;

            if (code.size() == 0)
            {
                status_code = -1; // 代码为空
                goto END;
            }

            // 生成源文件并写入
            file_name = FileUtil::UniqueFileName();
            if (!FileUtil::WriteFile(PathUtil::Src(file_name), code))
            {
                status_code = -2; // 文件写入错误
                goto END;
            }

            // 编译源文件
            if (!Compiler::Compile(file_name))
            {
                status_code = -3; // 代码编译错误
                goto END;
            }

            // 运行可执行文件
            ret_code = Runner::Run(file_name, cpu_limit, mem_limit);
            if (ret_code < 0) // 内部错误
            {
                status_code = -2;
                goto END;
            }
            else // ret_code >= 0 程序异常或正常退出
            {
                status_code = ret_code;
                goto END;
            }

        END:
            Json::Value out_value;
            std::string stdout_content;
            std::string stderr_content;

            out_value["status"] = status_code;
            out_value["reason"] = StatusUtil::CodeToDesc(status_code, file_name);

            if (status_code == 0)
            {
                FileUtil::ReadFile(PathUtil::Stdout(file_name), &stdout_content, true);
                FileUtil::ReadFile(PathUtil::Stderr(file_name), &stderr_content, true);
                out_value["stdout"] = stdout_content;
                out_value["stderr"] = stderr_content;
            }

            //序列化
            Json::StyledWriter writer;
            *out_json = writer.write(out_value);

            // FileUtil::DeleteTempFile(file_name); // 清理临时文件

        }
    };

}
